{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "from autogluon.tabular import TabularDataset, TabularPredictor\n",
    "from sklearn.metrics import accuracy_score, confusion_matrix, classification_report\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.preprocessing import LabelEncoder\n",
    "from sklearn.preprocessing import StandardScaler"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "train_data = pd.read_excel('q1_concat1.xlsx')\n",
    "train_label = pd.read_excel(r'数据\\竞赛发布数据\\表4-答案文件更新后.xlsx',skiprows=2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "label = train_label['1是，0否']\n",
    "label_train = label[:100]\n",
    "label_test = label[100:]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "23"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "(label == 1).sum()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "feature = train_data.iloc[:,4:]\n",
    "feature_train = feature[:100]\n",
    "feature_test = feature[100:]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>年龄</th>\n",
       "      <th>性别</th>\n",
       "      <th>脑出血前mRS评分</th>\n",
       "      <th>高血压病史</th>\n",
       "      <th>卒中病史</th>\n",
       "      <th>糖尿病史</th>\n",
       "      <th>房颤史</th>\n",
       "      <th>冠心病史</th>\n",
       "      <th>吸烟史</th>\n",
       "      <th>饮酒史</th>\n",
       "      <th>...</th>\n",
       "      <th>NCCT_original_firstorder_Median_y</th>\n",
       "      <th>NCCT_original_firstorder_Minimum_y</th>\n",
       "      <th>NCCT_original_firstorder_Range_y</th>\n",
       "      <th>NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y</th>\n",
       "      <th>NCCT_original_firstorder_RootMeanSquared_y</th>\n",
       "      <th>NCCT_original_firstorder_Skewness_y</th>\n",
       "      <th>NCCT_original_firstorder_Uniformity_y</th>\n",
       "      <th>NCCT_original_firstorder_Variance_y</th>\n",
       "      <th>高压</th>\n",
       "      <th>低压</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>43</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>54.277975</td>\n",
       "      <td>2.630186</td>\n",
       "      <td>88.333203</td>\n",
       "      <td>5.421960</td>\n",
       "      <td>54.182743</td>\n",
       "      <td>-0.680312</td>\n",
       "      <td>0.156607</td>\n",
       "      <td>116.141179</td>\n",
       "      <td>180</td>\n",
       "      <td>90</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>58</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>37.518627</td>\n",
       "      <td>11.670399</td>\n",
       "      <td>54.126259</td>\n",
       "      <td>3.986374</td>\n",
       "      <td>37.958520</td>\n",
       "      <td>-0.176969</td>\n",
       "      <td>0.132046</td>\n",
       "      <td>53.076248</td>\n",
       "      <td>199</td>\n",
       "      <td>120</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>78</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>35.410096</td>\n",
       "      <td>-1.715890</td>\n",
       "      <td>70.139106</td>\n",
       "      <td>5.382112</td>\n",
       "      <td>36.444232</td>\n",
       "      <td>-0.057096</td>\n",
       "      <td>0.128060</td>\n",
       "      <td>91.740851</td>\n",
       "      <td>199</td>\n",
       "      <td>120</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>70</td>\n",
       "      <td>男</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>30.926762</td>\n",
       "      <td>-20.991359</td>\n",
       "      <td>92.048538</td>\n",
       "      <td>6.191501</td>\n",
       "      <td>32.204819</td>\n",
       "      <td>-0.297133</td>\n",
       "      <td>0.145292</td>\n",
       "      <td>125.651157</td>\n",
       "      <td>186</td>\n",
       "      <td>99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>51</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>46.492200</td>\n",
       "      <td>-9.889141</td>\n",
       "      <td>96.821808</td>\n",
       "      <td>6.482624</td>\n",
       "      <td>47.428035</td>\n",
       "      <td>-0.227378</td>\n",
       "      <td>0.146488</td>\n",
       "      <td>134.720581</td>\n",
       "      <td>135</td>\n",
       "      <td>92</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>155</th>\n",
       "      <td>87</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>31.471558</td>\n",
       "      <td>-19.978141</td>\n",
       "      <td>111.297179</td>\n",
       "      <td>7.310788</td>\n",
       "      <td>33.601480</td>\n",
       "      <td>-0.068704</td>\n",
       "      <td>0.151482</td>\n",
       "      <td>169.061898</td>\n",
       "      <td>202</td>\n",
       "      <td>100</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>156</th>\n",
       "      <td>52</td>\n",
       "      <td>男</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>48.533011</td>\n",
       "      <td>30.460776</td>\n",
       "      <td>29.788433</td>\n",
       "      <td>2.260258</td>\n",
       "      <td>48.602582</td>\n",
       "      <td>-0.239727</td>\n",
       "      <td>0.125220</td>\n",
       "      <td>24.684739</td>\n",
       "      <td>180</td>\n",
       "      <td>110</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>157</th>\n",
       "      <td>57</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>53.345998</td>\n",
       "      <td>20.456862</td>\n",
       "      <td>65.707611</td>\n",
       "      <td>4.718427</td>\n",
       "      <td>53.690717</td>\n",
       "      <td>-0.079589</td>\n",
       "      <td>0.136352</td>\n",
       "      <td>73.002568</td>\n",
       "      <td>233</td>\n",
       "      <td>135</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>158</th>\n",
       "      <td>47</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>37.092745</td>\n",
       "      <td>-5.551496</td>\n",
       "      <td>72.713624</td>\n",
       "      <td>5.802628</td>\n",
       "      <td>38.494810</td>\n",
       "      <td>-0.202350</td>\n",
       "      <td>0.120587</td>\n",
       "      <td>114.191236</td>\n",
       "      <td>183</td>\n",
       "      <td>122</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>159</th>\n",
       "      <td>80</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>44.975763</td>\n",
       "      <td>7.592122</td>\n",
       "      <td>71.920477</td>\n",
       "      <td>5.207770</td>\n",
       "      <td>45.857394</td>\n",
       "      <td>-0.149273</td>\n",
       "      <td>0.132971</td>\n",
       "      <td>92.463632</td>\n",
       "      <td>208</td>\n",
       "      <td>95</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>160 rows × 104 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     年龄 性别  脑出血前mRS评分  高血压病史  卒中病史  糖尿病史  房颤史  冠心病史  吸烟史  饮酒史  ...  \\\n",
       "0    43  女          0      0     0     0    0     0    0    0  ...   \n",
       "1    58  男          0      1     0     0    0     0    0    0  ...   \n",
       "2    78  男          0      1     0     0    0     0    0    0  ...   \n",
       "3    70  男          2      1     1     0    0     0    0    0  ...   \n",
       "4    51  男          0      0     0     0    0     0    0    0  ...   \n",
       "..   .. ..        ...    ...   ...   ...  ...   ...  ...  ...  ...   \n",
       "155  87  女          0      1     0     0    0     1    0    0  ...   \n",
       "156  52  男          2      1     1     0    0     0    1    0  ...   \n",
       "157  57  男          0      1     0     0    0     0    0    0  ...   \n",
       "158  47  男          0      1     0     0    0     0    1    1  ...   \n",
       "159  80  女          0      1     0     1    0     0    0    0  ...   \n",
       "\n",
       "     NCCT_original_firstorder_Median_y  NCCT_original_firstorder_Minimum_y  \\\n",
       "0                            54.277975                            2.630186   \n",
       "1                            37.518627                           11.670399   \n",
       "2                            35.410096                           -1.715890   \n",
       "3                            30.926762                          -20.991359   \n",
       "4                            46.492200                           -9.889141   \n",
       "..                                 ...                                 ...   \n",
       "155                          31.471558                          -19.978141   \n",
       "156                          48.533011                           30.460776   \n",
       "157                          53.345998                           20.456862   \n",
       "158                          37.092745                           -5.551496   \n",
       "159                          44.975763                            7.592122   \n",
       "\n",
       "     NCCT_original_firstorder_Range_y  \\\n",
       "0                           88.333203   \n",
       "1                           54.126259   \n",
       "2                           70.139106   \n",
       "3                           92.048538   \n",
       "4                           96.821808   \n",
       "..                                ...   \n",
       "155                        111.297179   \n",
       "156                         29.788433   \n",
       "157                         65.707611   \n",
       "158                         72.713624   \n",
       "159                         71.920477   \n",
       "\n",
       "     NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y  \\\n",
       "0                                             5.421960        \n",
       "1                                             3.986374        \n",
       "2                                             5.382112        \n",
       "3                                             6.191501        \n",
       "4                                             6.482624        \n",
       "..                                                 ...        \n",
       "155                                           7.310788        \n",
       "156                                           2.260258        \n",
       "157                                           4.718427        \n",
       "158                                           5.802628        \n",
       "159                                           5.207770        \n",
       "\n",
       "     NCCT_original_firstorder_RootMeanSquared_y  \\\n",
       "0                                     54.182743   \n",
       "1                                     37.958520   \n",
       "2                                     36.444232   \n",
       "3                                     32.204819   \n",
       "4                                     47.428035   \n",
       "..                                          ...   \n",
       "155                                   33.601480   \n",
       "156                                   48.602582   \n",
       "157                                   53.690717   \n",
       "158                                   38.494810   \n",
       "159                                   45.857394   \n",
       "\n",
       "     NCCT_original_firstorder_Skewness_y  \\\n",
       "0                              -0.680312   \n",
       "1                              -0.176969   \n",
       "2                              -0.057096   \n",
       "3                              -0.297133   \n",
       "4                              -0.227378   \n",
       "..                                   ...   \n",
       "155                            -0.068704   \n",
       "156                            -0.239727   \n",
       "157                            -0.079589   \n",
       "158                            -0.202350   \n",
       "159                            -0.149273   \n",
       "\n",
       "     NCCT_original_firstorder_Uniformity_y  \\\n",
       "0                                 0.156607   \n",
       "1                                 0.132046   \n",
       "2                                 0.128060   \n",
       "3                                 0.145292   \n",
       "4                                 0.146488   \n",
       "..                                     ...   \n",
       "155                               0.151482   \n",
       "156                               0.125220   \n",
       "157                               0.136352   \n",
       "158                               0.120587   \n",
       "159                               0.132971   \n",
       "\n",
       "     NCCT_original_firstorder_Variance_y   高压   低压  \n",
       "0                             116.141179  180   90  \n",
       "1                              53.076248  199  120  \n",
       "2                              91.740851  199  120  \n",
       "3                             125.651157  186   99  \n",
       "4                             134.720581  135   92  \n",
       "..                                   ...  ...  ...  \n",
       "155                           169.061898  202  100  \n",
       "156                            24.684739  180  110  \n",
       "157                            73.002568  233  135  \n",
       "158                           114.191236  183  122  \n",
       "159                            92.463632  208   95  \n",
       "\n",
       "[160 rows x 104 columns]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "feature"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "No path specified. Models will be saved in: \"AutogluonModels\\ag-20230923_044111\\\"\n",
      "Beginning AutoGluon training ...\n",
      "AutoGluon will save models to \"AutogluonModels\\ag-20230923_044111\\\"\n",
      "AutoGluon Version:  0.7.0\n",
      "Python Version:     3.9.12\n",
      "Operating System:   Windows\n",
      "Platform Machine:   AMD64\n",
      "Platform Version:   10.0.22621\n",
      "Train Data Rows:    146\n",
      "Train Data Columns: 104\n",
      "Label Column: class\n",
      "Preprocessing data ...\n",
      "AutoGluon infers your prediction problem is: 'binary' (because only two unique label-values observed).\n",
      "\t2 unique label values:  [0.0, 1.0]\n",
      "\tIf 'binary' is not the correct problem_type, please manually specify the problem_type parameter during predictor init (You may specify problem_type as one of: ['binary', 'multiclass', 'regression'])\n",
      "Selected class <--> label mapping:  class 1 = 1, class 0 = 0\n",
      "Using Feature Generators to preprocess the data ...\n",
      "Fitting AutoMLPipelineFeatureGenerator...\n",
      "\tAvailable Memory:                    4534.83 MB\n",
      "\tTrain Data (Original)  Memory Usage: 0.13 MB (0.0% of available memory)\n",
      "\tInferring data type of each feature based on column values. Set feature_metadata_in to manually specify special dtypes of the features.\n",
      "\tStage 1 Generators:\n",
      "\t\tFitting AsTypeFeatureGenerator...\n",
      "\t\t\tNote: Converting 15 features to boolean dtype as they only contain 2 unique values.\n",
      "\tStage 2 Generators:\n",
      "\t\tFitting FillNaFeatureGenerator...\n",
      "\tStage 3 Generators:\n",
      "\t\tFitting IdentityFeatureGenerator...\n",
      "\tStage 4 Generators:\n",
      "\t\tFitting DropUniqueFeatureGenerator...\n",
      "\tTypes of features in original data (raw dtype, special dtypes):\n",
      "\t\t('float', [])  : 81 | ['发病到首次影像检查时间间隔', 'HM_ACA_R_Ratio', 'HM_MCA_R_Ratio', 'HM_PCA_R_Ratio', 'HM_Pons_Medulla_R_Ratio', ...]\n",
      "\t\t('int', [])    : 22 | ['年龄', '脑出血前mRS评分', '高血压病史', '卒中病史', '糖尿病史', ...]\n",
      "\t\t('object', []) :  1 | ['性别']\n",
      "\tTypes of features in processed data (raw dtype, special dtypes):\n",
      "\t\t('float', [])     : 81 | ['发病到首次影像检查时间间隔', 'HM_ACA_R_Ratio', 'HM_MCA_R_Ratio', 'HM_PCA_R_Ratio', 'HM_Pons_Medulla_R_Ratio', ...]\n",
      "\t\t('int', [])       :  8 | ['年龄', '脑出血前mRS评分', 'HM_volume', 'ED_volume', 'original_shape_VoxelVolume_x', ...]\n",
      "\t\t('int', ['bool']) : 15 | ['性别', '高血压病史', '卒中病史', '糖尿病史', '房颤史', ...]\n",
      "\t0.1s = Fit runtime\n",
      "\t104 features in original data used to generate 104 features in processed data.\n",
      "\tTrain Data (Processed) Memory Usage: 0.11 MB (0.0% of available memory)\n",
      "Data preprocessing and feature engineering runtime = 0.08s ...\n",
      "AutoGluon will gauge predictive performance using evaluation metric: 'accuracy'\n",
      "\tTo change this, specify the eval_metric parameter of Predictor()\n",
      "Automatically generating train/validation split with holdout_frac=0.2, Train Rows: 116, Val Rows: 30\n",
      "Fitting 13 L1 models ...\n",
      "Fitting model: KNeighborsUnif ...\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\t0.3667\t = Validation score   (accuracy)\n",
      "\t2.22s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: KNeighborsDist ...\n",
      "\t0.6333\t = Validation score   (accuracy)\n",
      "\t0.01s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: LightGBMXT ...\n",
      "\t0.8667\t = Validation score   (accuracy)\n",
      "\t0.69s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: LightGBM ...\n",
      "\t0.8667\t = Validation score   (accuracy)\n",
      "\t0.2s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: RandomForestGini ...\n",
      "\t0.9\t = Validation score   (accuracy)\n",
      "\t0.42s\t = Training   runtime\n",
      "\t0.05s\t = Validation runtime\n",
      "Fitting model: RandomForestEntr ...\n",
      "\t0.8667\t = Validation score   (accuracy)\n",
      "\t0.4s\t = Training   runtime\n",
      "\t0.04s\t = Validation runtime\n",
      "Fitting model: CatBoost ...\n",
      "\t0.9\t = Validation score   (accuracy)\n",
      "\t1.1s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: ExtraTreesGini ...\n",
      "\t0.9\t = Validation score   (accuracy)\n",
      "\t0.32s\t = Training   runtime\n",
      "\t0.04s\t = Validation runtime\n",
      "Fitting model: ExtraTreesEntr ...\n",
      "\t0.8667\t = Validation score   (accuracy)\n",
      "\t0.32s\t = Training   runtime\n",
      "\t0.05s\t = Validation runtime\n",
      "Fitting model: NeuralNetFastAI ...\n",
      "No improvement since epoch 5: early stopping\n",
      "\t0.9\t = Validation score   (accuracy)\n",
      "\t1.82s\t = Training   runtime\n",
      "\t0.01s\t = Validation runtime\n",
      "Fitting model: XGBoost ...\n",
      "\t0.8667\t = Validation score   (accuracy)\n",
      "\t0.18s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: NeuralNetTorch ...\n",
      "\t0.9\t = Validation score   (accuracy)\n",
      "\t0.91s\t = Training   runtime\n",
      "\t0.01s\t = Validation runtime\n",
      "Fitting model: LightGBMLarge ...\n",
      "\t0.9667\t = Validation score   (accuracy)\n",
      "\t0.26s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "Fitting model: WeightedEnsemble_L2 ...\n",
      "\t0.9667\t = Validation score   (accuracy)\n",
      "\t0.18s\t = Training   runtime\n",
      "\t0.0s\t = Validation runtime\n",
      "AutoGluon training complete, total runtime = 9.5s ... Best model: \"WeightedEnsemble_L2\"\n",
      "TabularPredictor saved. To load, use: predictor = TabularPredictor.load(\"AutogluonModels\\ag-20230923_044111\\\")\n"
     ]
    }
   ],
   "source": [
    "import random\n",
    "train_set = pd.concat([feature_train,label_train],axis=1)\n",
    "test_set = pd.concat([feature_test],axis=1)\n",
    "label_name = train_set.columns[-1]\n",
    "raw_trian = train_set.rename(columns={label_name:'class'})\n",
    "train_set_1 = raw_trian[raw_trian['class']==1]\n",
    "# repeat 5 times\n",
    "train_set = pd.concat([raw_trian,train_set_1,train_set_1],axis=0)\n",
    "train_set = train_set.sample(frac=1)\n",
    "# autoML\n",
    "train_data = TabularDataset(train_set)\n",
    "\n",
    "predictor = TabularPredictor(label='class').fit(train_data)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>年龄</th>\n",
       "      <th>性别</th>\n",
       "      <th>脑出血前mRS评分</th>\n",
       "      <th>高血压病史</th>\n",
       "      <th>卒中病史</th>\n",
       "      <th>糖尿病史</th>\n",
       "      <th>房颤史</th>\n",
       "      <th>冠心病史</th>\n",
       "      <th>吸烟史</th>\n",
       "      <th>饮酒史</th>\n",
       "      <th>...</th>\n",
       "      <th>NCCT_original_firstorder_Minimum_y</th>\n",
       "      <th>NCCT_original_firstorder_Range_y</th>\n",
       "      <th>NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y</th>\n",
       "      <th>NCCT_original_firstorder_RootMeanSquared_y</th>\n",
       "      <th>NCCT_original_firstorder_Skewness_y</th>\n",
       "      <th>NCCT_original_firstorder_Uniformity_y</th>\n",
       "      <th>NCCT_original_firstorder_Variance_y</th>\n",
       "      <th>高压</th>\n",
       "      <th>低压</th>\n",
       "      <th>class</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>81</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.208050</td>\n",
       "      <td>78.398870</td>\n",
       "      <td>5.016021</td>\n",
       "      <td>55.579146</td>\n",
       "      <td>-0.137280</td>\n",
       "      <td>0.150416</td>\n",
       "      <td>89.223067</td>\n",
       "      <td>170</td>\n",
       "      <td>100</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>53</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.340002</td>\n",
       "      <td>76.238558</td>\n",
       "      <td>4.513306</td>\n",
       "      <td>55.558740</td>\n",
       "      <td>0.103995</td>\n",
       "      <td>0.163955</td>\n",
       "      <td>67.511023</td>\n",
       "      <td>138</td>\n",
       "      <td>84</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>91</th>\n",
       "      <td>86</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>8.389589</td>\n",
       "      <td>44.480282</td>\n",
       "      <td>3.825139</td>\n",
       "      <td>34.974667</td>\n",
       "      <td>-0.252667</td>\n",
       "      <td>0.114170</td>\n",
       "      <td>67.302729</td>\n",
       "      <td>155</td>\n",
       "      <td>76</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>69</th>\n",
       "      <td>84</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.876295</td>\n",
       "      <td>81.855433</td>\n",
       "      <td>6.434509</td>\n",
       "      <td>42.016589</td>\n",
       "      <td>-0.193205</td>\n",
       "      <td>0.124880</td>\n",
       "      <td>136.783129</td>\n",
       "      <td>150</td>\n",
       "      <td>80</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>72</th>\n",
       "      <td>52</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>6.127457</td>\n",
       "      <td>72.807811</td>\n",
       "      <td>5.353694</td>\n",
       "      <td>48.241905</td>\n",
       "      <td>-0.177122</td>\n",
       "      <td>0.132118</td>\n",
       "      <td>96.547872</td>\n",
       "      <td>141</td>\n",
       "      <td>96</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>79</th>\n",
       "      <td>68</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>14.224362</td>\n",
       "      <td>68.359928</td>\n",
       "      <td>5.695636</td>\n",
       "      <td>53.880508</td>\n",
       "      <td>-0.494925</td>\n",
       "      <td>0.117709</td>\n",
       "      <td>113.413878</td>\n",
       "      <td>178</td>\n",
       "      <td>87</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>81</th>\n",
       "      <td>48</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0.412895</td>\n",
       "      <td>82.078431</td>\n",
       "      <td>4.650560</td>\n",
       "      <td>51.984249</td>\n",
       "      <td>-0.577688</td>\n",
       "      <td>0.171539</td>\n",
       "      <td>79.901431</td>\n",
       "      <td>240</td>\n",
       "      <td>110</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>62</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-4.031646</td>\n",
       "      <td>69.452036</td>\n",
       "      <td>3.976634</td>\n",
       "      <td>39.931655</td>\n",
       "      <td>-0.270221</td>\n",
       "      <td>0.166303</td>\n",
       "      <td>58.026317</td>\n",
       "      <td>167</td>\n",
       "      <td>117</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>97</th>\n",
       "      <td>82</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.467873</td>\n",
       "      <td>98.855412</td>\n",
       "      <td>5.523299</td>\n",
       "      <td>37.597608</td>\n",
       "      <td>0.432499</td>\n",
       "      <td>0.166861</td>\n",
       "      <td>123.960058</td>\n",
       "      <td>200</td>\n",
       "      <td>100</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>82</th>\n",
       "      <td>45</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-25.536902</td>\n",
       "      <td>104.103838</td>\n",
       "      <td>7.246197</td>\n",
       "      <td>29.951308</td>\n",
       "      <td>-0.224427</td>\n",
       "      <td>0.139686</td>\n",
       "      <td>179.115822</td>\n",
       "      <td>170</td>\n",
       "      <td>90</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>146 rows × 105 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    年龄 性别  脑出血前mRS评分  高血压病史  卒中病史  糖尿病史  房颤史  冠心病史  吸烟史  饮酒史  ...  \\\n",
       "14  81  男          0      1     1     1    0     0    1    0  ...   \n",
       "25  53  女          0      1     0     0    0     0    0    0  ...   \n",
       "91  86  男          0      1     1     1    0     1    0    0  ...   \n",
       "69  84  女          0      1     1     0    0     0    0    0  ...   \n",
       "72  52  男          0      1     0     0    0     0    0    0  ...   \n",
       "..  .. ..        ...    ...   ...   ...  ...   ...  ...  ...  ...   \n",
       "79  68  男          0      1     0     0    0     0    1    0  ...   \n",
       "81  48  男          0      1     0     0    0     0    1    1  ...   \n",
       "16  62  男          0      1     1     0    0     0    0    0  ...   \n",
       "97  82  男          0      1     1     1    1     1    1    0  ...   \n",
       "82  45  男          0      1     0     1    0     0    1    0  ...   \n",
       "\n",
       "    NCCT_original_firstorder_Minimum_y  NCCT_original_firstorder_Range_y  \\\n",
       "14                           15.208050                         78.398870   \n",
       "25                           13.340002                         76.238558   \n",
       "91                            8.389589                         44.480282   \n",
       "69                           -2.876295                         81.855433   \n",
       "72                            6.127457                         72.807811   \n",
       "..                                 ...                               ...   \n",
       "79                           14.224362                         68.359928   \n",
       "81                            0.412895                         82.078431   \n",
       "16                           -4.031646                         69.452036   \n",
       "97                           -2.467873                         98.855412   \n",
       "82                          -25.536902                        104.103838   \n",
       "\n",
       "    NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y  \\\n",
       "14                                                5.016021   \n",
       "25                                                4.513306   \n",
       "91                                                3.825139   \n",
       "69                                                6.434509   \n",
       "72                                                5.353694   \n",
       "..                                                     ...   \n",
       "79                                                5.695636   \n",
       "81                                                4.650560   \n",
       "16                                                3.976634   \n",
       "97                                                5.523299   \n",
       "82                                                7.246197   \n",
       "\n",
       "    NCCT_original_firstorder_RootMeanSquared_y  \\\n",
       "14                                   55.579146   \n",
       "25                                   55.558740   \n",
       "91                                   34.974667   \n",
       "69                                   42.016589   \n",
       "72                                   48.241905   \n",
       "..                                         ...   \n",
       "79                                   53.880508   \n",
       "81                                   51.984249   \n",
       "16                                   39.931655   \n",
       "97                                   37.597608   \n",
       "82                                   29.951308   \n",
       "\n",
       "    NCCT_original_firstorder_Skewness_y  \\\n",
       "14                            -0.137280   \n",
       "25                             0.103995   \n",
       "91                            -0.252667   \n",
       "69                            -0.193205   \n",
       "72                            -0.177122   \n",
       "..                                  ...   \n",
       "79                            -0.494925   \n",
       "81                            -0.577688   \n",
       "16                            -0.270221   \n",
       "97                             0.432499   \n",
       "82                            -0.224427   \n",
       "\n",
       "    NCCT_original_firstorder_Uniformity_y  \\\n",
       "14                               0.150416   \n",
       "25                               0.163955   \n",
       "91                               0.114170   \n",
       "69                               0.124880   \n",
       "72                               0.132118   \n",
       "..                                    ...   \n",
       "79                               0.117709   \n",
       "81                               0.171539   \n",
       "16                               0.166303   \n",
       "97                               0.166861   \n",
       "82                               0.139686   \n",
       "\n",
       "    NCCT_original_firstorder_Variance_y   高压   低压  class  \n",
       "14                            89.223067  170  100    0.0  \n",
       "25                            67.511023  138   84    0.0  \n",
       "91                            67.302729  155   76    1.0  \n",
       "69                           136.783129  150   80    1.0  \n",
       "72                            96.547872  141   96    0.0  \n",
       "..                                  ...  ...  ...    ...  \n",
       "79                           113.413878  178   87    1.0  \n",
       "81                            79.901431  240  110    0.0  \n",
       "16                            58.026317  167  117    1.0  \n",
       "97                           123.960058  200  100    1.0  \n",
       "82                           179.115822  170   90    0.0  \n",
       "\n",
       "[146 rows x 105 columns]"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_set"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_label = predictor.predict(test_set)\n",
    "test_prob = predictor.predict_proba(test_set)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "leaderboard = predictor.leaderboard(silent=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "leaderboard.to_excel('q1_b_leaderboard.xlsx')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "test_set['class'] = test_label"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "acc_train = accuracy_score(predictor.predict(train_set),train_set['class'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "total_feature = pd.read_excel('q1_concat1.xlsx')\n",
    "prob = predictor.predict_proba(total_feature)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "prob.to_excel('q1_b_prob.xlsx')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Computing feature importance via permutation shuffling for 104 features using 100 rows with 5 shuffle sets...\n",
      "\t6.58s\t= Expected runtime (1.32s per shuffle set)\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\t0.7s\t= Actual runtime (Completed 5 of 5 shuffle sets)\n"
     ]
    }
   ],
   "source": [
    "feature_importance = predictor.feature_importance(raw_trian)\n",
    "feature_importance.to_excel('q1_b_feature_importance.xlsx')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "acc = accuracy_score(predictor.predict(raw_trian),raw_trian['class'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.99"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "acc"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "145"
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "(predictor.predict(train_data) == train_data['class']).sum()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>年龄</th>\n",
       "      <th>性别</th>\n",
       "      <th>脑出血前mRS评分</th>\n",
       "      <th>高血压病史</th>\n",
       "      <th>卒中病史</th>\n",
       "      <th>糖尿病史</th>\n",
       "      <th>房颤史</th>\n",
       "      <th>冠心病史</th>\n",
       "      <th>吸烟史</th>\n",
       "      <th>饮酒史</th>\n",
       "      <th>...</th>\n",
       "      <th>NCCT_original_firstorder_Minimum_y</th>\n",
       "      <th>NCCT_original_firstorder_Range_y</th>\n",
       "      <th>NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y</th>\n",
       "      <th>NCCT_original_firstorder_RootMeanSquared_y</th>\n",
       "      <th>NCCT_original_firstorder_Skewness_y</th>\n",
       "      <th>NCCT_original_firstorder_Uniformity_y</th>\n",
       "      <th>NCCT_original_firstorder_Variance_y</th>\n",
       "      <th>高压</th>\n",
       "      <th>低压</th>\n",
       "      <th>class</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>81</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>15.208050</td>\n",
       "      <td>78.398870</td>\n",
       "      <td>5.016021</td>\n",
       "      <td>55.579146</td>\n",
       "      <td>-0.137280</td>\n",
       "      <td>0.150416</td>\n",
       "      <td>89.223067</td>\n",
       "      <td>170</td>\n",
       "      <td>100</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>53</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>13.340002</td>\n",
       "      <td>76.238558</td>\n",
       "      <td>4.513306</td>\n",
       "      <td>55.558740</td>\n",
       "      <td>0.103995</td>\n",
       "      <td>0.163955</td>\n",
       "      <td>67.511023</td>\n",
       "      <td>138</td>\n",
       "      <td>84</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>91</th>\n",
       "      <td>86</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>8.389589</td>\n",
       "      <td>44.480282</td>\n",
       "      <td>3.825139</td>\n",
       "      <td>34.974667</td>\n",
       "      <td>-0.252667</td>\n",
       "      <td>0.114170</td>\n",
       "      <td>67.302729</td>\n",
       "      <td>155</td>\n",
       "      <td>76</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>69</th>\n",
       "      <td>84</td>\n",
       "      <td>女</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.876295</td>\n",
       "      <td>81.855433</td>\n",
       "      <td>6.434509</td>\n",
       "      <td>42.016589</td>\n",
       "      <td>-0.193205</td>\n",
       "      <td>0.124880</td>\n",
       "      <td>136.783129</td>\n",
       "      <td>150</td>\n",
       "      <td>80</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>72</th>\n",
       "      <td>52</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>6.127457</td>\n",
       "      <td>72.807811</td>\n",
       "      <td>5.353694</td>\n",
       "      <td>48.241905</td>\n",
       "      <td>-0.177122</td>\n",
       "      <td>0.132118</td>\n",
       "      <td>96.547872</td>\n",
       "      <td>141</td>\n",
       "      <td>96</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>79</th>\n",
       "      <td>68</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>14.224362</td>\n",
       "      <td>68.359928</td>\n",
       "      <td>5.695636</td>\n",
       "      <td>53.880508</td>\n",
       "      <td>-0.494925</td>\n",
       "      <td>0.117709</td>\n",
       "      <td>113.413878</td>\n",
       "      <td>178</td>\n",
       "      <td>87</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>81</th>\n",
       "      <td>48</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0.412895</td>\n",
       "      <td>82.078431</td>\n",
       "      <td>4.650560</td>\n",
       "      <td>51.984249</td>\n",
       "      <td>-0.577688</td>\n",
       "      <td>0.171539</td>\n",
       "      <td>79.901431</td>\n",
       "      <td>240</td>\n",
       "      <td>110</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>62</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-4.031646</td>\n",
       "      <td>69.452036</td>\n",
       "      <td>3.976634</td>\n",
       "      <td>39.931655</td>\n",
       "      <td>-0.270221</td>\n",
       "      <td>0.166303</td>\n",
       "      <td>58.026317</td>\n",
       "      <td>167</td>\n",
       "      <td>117</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>97</th>\n",
       "      <td>82</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-2.467873</td>\n",
       "      <td>98.855412</td>\n",
       "      <td>5.523299</td>\n",
       "      <td>37.597608</td>\n",
       "      <td>0.432499</td>\n",
       "      <td>0.166861</td>\n",
       "      <td>123.960058</td>\n",
       "      <td>200</td>\n",
       "      <td>100</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>82</th>\n",
       "      <td>45</td>\n",
       "      <td>男</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>-25.536902</td>\n",
       "      <td>104.103838</td>\n",
       "      <td>7.246197</td>\n",
       "      <td>29.951308</td>\n",
       "      <td>-0.224427</td>\n",
       "      <td>0.139686</td>\n",
       "      <td>179.115822</td>\n",
       "      <td>170</td>\n",
       "      <td>90</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>146 rows × 105 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "    年龄 性别  脑出血前mRS评分  高血压病史  卒中病史  糖尿病史  房颤史  冠心病史  吸烟史  饮酒史  ...  \\\n",
       "14  81  男          0      1     1     1    0     0    1    0  ...   \n",
       "25  53  女          0      1     0     0    0     0    0    0  ...   \n",
       "91  86  男          0      1     1     1    0     1    0    0  ...   \n",
       "69  84  女          0      1     1     0    0     0    0    0  ...   \n",
       "72  52  男          0      1     0     0    0     0    0    0  ...   \n",
       "..  .. ..        ...    ...   ...   ...  ...   ...  ...  ...  ...   \n",
       "79  68  男          0      1     0     0    0     0    1    0  ...   \n",
       "81  48  男          0      1     0     0    0     0    1    1  ...   \n",
       "16  62  男          0      1     1     0    0     0    0    0  ...   \n",
       "97  82  男          0      1     1     1    1     1    1    0  ...   \n",
       "82  45  男          0      1     0     1    0     0    1    0  ...   \n",
       "\n",
       "    NCCT_original_firstorder_Minimum_y  NCCT_original_firstorder_Range_y  \\\n",
       "14                           15.208050                         78.398870   \n",
       "25                           13.340002                         76.238558   \n",
       "91                            8.389589                         44.480282   \n",
       "69                           -2.876295                         81.855433   \n",
       "72                            6.127457                         72.807811   \n",
       "..                                 ...                               ...   \n",
       "79                           14.224362                         68.359928   \n",
       "81                            0.412895                         82.078431   \n",
       "16                           -4.031646                         69.452036   \n",
       "97                           -2.467873                         98.855412   \n",
       "82                          -25.536902                        104.103838   \n",
       "\n",
       "    NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y  \\\n",
       "14                                                5.016021   \n",
       "25                                                4.513306   \n",
       "91                                                3.825139   \n",
       "69                                                6.434509   \n",
       "72                                                5.353694   \n",
       "..                                                     ...   \n",
       "79                                                5.695636   \n",
       "81                                                4.650560   \n",
       "16                                                3.976634   \n",
       "97                                                5.523299   \n",
       "82                                                7.246197   \n",
       "\n",
       "    NCCT_original_firstorder_RootMeanSquared_y  \\\n",
       "14                                   55.579146   \n",
       "25                                   55.558740   \n",
       "91                                   34.974667   \n",
       "69                                   42.016589   \n",
       "72                                   48.241905   \n",
       "..                                         ...   \n",
       "79                                   53.880508   \n",
       "81                                   51.984249   \n",
       "16                                   39.931655   \n",
       "97                                   37.597608   \n",
       "82                                   29.951308   \n",
       "\n",
       "    NCCT_original_firstorder_Skewness_y  \\\n",
       "14                            -0.137280   \n",
       "25                             0.103995   \n",
       "91                            -0.252667   \n",
       "69                            -0.193205   \n",
       "72                            -0.177122   \n",
       "..                                  ...   \n",
       "79                            -0.494925   \n",
       "81                            -0.577688   \n",
       "16                            -0.270221   \n",
       "97                             0.432499   \n",
       "82                            -0.224427   \n",
       "\n",
       "    NCCT_original_firstorder_Uniformity_y  \\\n",
       "14                               0.150416   \n",
       "25                               0.163955   \n",
       "91                               0.114170   \n",
       "69                               0.124880   \n",
       "72                               0.132118   \n",
       "..                                    ...   \n",
       "79                               0.117709   \n",
       "81                               0.171539   \n",
       "16                               0.166303   \n",
       "97                               0.166861   \n",
       "82                               0.139686   \n",
       "\n",
       "    NCCT_original_firstorder_Variance_y   高压   低压  class  \n",
       "14                            89.223067  170  100    0.0  \n",
       "25                            67.511023  138   84    0.0  \n",
       "91                            67.302729  155   76    1.0  \n",
       "69                           136.783129  150   80    1.0  \n",
       "72                            96.547872  141   96    0.0  \n",
       "..                                  ...  ...  ...    ...  \n",
       "79                           113.413878  178   87    1.0  \n",
       "81                            79.901431  240  110    0.0  \n",
       "16                            58.026317  167  117    1.0  \n",
       "97                           123.960058  200  100    1.0  \n",
       "82                           179.115822  170   90    0.0  \n",
       "\n",
       "[146 rows x 105 columns]"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "                  model  score_val  pred_time_val  fit_time  pred_time_val_marginal  fit_time_marginal  stack_level  can_infer  fit_order\n",
      "0         LightGBMLarge   0.966667       0.002011  0.261748                0.002011           0.261748            1       True         13\n",
      "1   WeightedEnsemble_L2   0.966667       0.002989  0.440563                0.000978           0.178815            2       True         14\n",
      "2              CatBoost   0.900000       0.003466  1.098149                0.003466           1.098149            1       True          7\n",
      "3       NeuralNetFastAI   0.900000       0.007771  1.824703                0.007771           1.824703            1       True         10\n",
      "4        NeuralNetTorch   0.900000       0.014041  0.907093                0.014041           0.907093            1       True         12\n",
      "5        ExtraTreesGini   0.900000       0.039440  0.316139                0.039440           0.316139            1       True          8\n",
      "6      RandomForestGini   0.900000       0.053123  0.424222                0.053123           0.424222            1       True          5\n",
      "7            LightGBMXT   0.866667       0.000998  0.694706                0.000998           0.694706            1       True          3\n",
      "8              LightGBM   0.866667       0.002006  0.196532                0.002006           0.196532            1       True          4\n",
      "9               XGBoost   0.866667       0.004032  0.182862                0.004032           0.182862            1       True         11\n",
      "10     RandomForestEntr   0.866667       0.039455  0.396508                0.039455           0.396508            1       True          6\n",
      "11       ExtraTreesEntr   0.866667       0.047879  0.319121                0.047879           0.319121            1       True          9\n",
      "12       KNeighborsDist   0.633333       0.000982  0.007527                0.000982           0.007527            1       True          2\n",
      "13       KNeighborsUnif   0.366667       0.001968  2.219379                0.001968           2.219379            1       True          1\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>model</th>\n",
       "      <th>score_val</th>\n",
       "      <th>pred_time_val</th>\n",
       "      <th>fit_time</th>\n",
       "      <th>pred_time_val_marginal</th>\n",
       "      <th>fit_time_marginal</th>\n",
       "      <th>stack_level</th>\n",
       "      <th>can_infer</th>\n",
       "      <th>fit_order</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>LightGBMLarge</td>\n",
       "      <td>0.966667</td>\n",
       "      <td>0.002011</td>\n",
       "      <td>0.261748</td>\n",
       "      <td>0.002011</td>\n",
       "      <td>0.261748</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>WeightedEnsemble_L2</td>\n",
       "      <td>0.966667</td>\n",
       "      <td>0.002989</td>\n",
       "      <td>0.440563</td>\n",
       "      <td>0.000978</td>\n",
       "      <td>0.178815</td>\n",
       "      <td>2</td>\n",
       "      <td>True</td>\n",
       "      <td>14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>CatBoost</td>\n",
       "      <td>0.900000</td>\n",
       "      <td>0.003466</td>\n",
       "      <td>1.098149</td>\n",
       "      <td>0.003466</td>\n",
       "      <td>1.098149</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>NeuralNetFastAI</td>\n",
       "      <td>0.900000</td>\n",
       "      <td>0.007771</td>\n",
       "      <td>1.824703</td>\n",
       "      <td>0.007771</td>\n",
       "      <td>1.824703</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>NeuralNetTorch</td>\n",
       "      <td>0.900000</td>\n",
       "      <td>0.014041</td>\n",
       "      <td>0.907093</td>\n",
       "      <td>0.014041</td>\n",
       "      <td>0.907093</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>ExtraTreesGini</td>\n",
       "      <td>0.900000</td>\n",
       "      <td>0.039440</td>\n",
       "      <td>0.316139</td>\n",
       "      <td>0.039440</td>\n",
       "      <td>0.316139</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>RandomForestGini</td>\n",
       "      <td>0.900000</td>\n",
       "      <td>0.053123</td>\n",
       "      <td>0.424222</td>\n",
       "      <td>0.053123</td>\n",
       "      <td>0.424222</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>LightGBMXT</td>\n",
       "      <td>0.866667</td>\n",
       "      <td>0.000998</td>\n",
       "      <td>0.694706</td>\n",
       "      <td>0.000998</td>\n",
       "      <td>0.694706</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>LightGBM</td>\n",
       "      <td>0.866667</td>\n",
       "      <td>0.002006</td>\n",
       "      <td>0.196532</td>\n",
       "      <td>0.002006</td>\n",
       "      <td>0.196532</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>XGBoost</td>\n",
       "      <td>0.866667</td>\n",
       "      <td>0.004032</td>\n",
       "      <td>0.182862</td>\n",
       "      <td>0.004032</td>\n",
       "      <td>0.182862</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>RandomForestEntr</td>\n",
       "      <td>0.866667</td>\n",
       "      <td>0.039455</td>\n",
       "      <td>0.396508</td>\n",
       "      <td>0.039455</td>\n",
       "      <td>0.396508</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>ExtraTreesEntr</td>\n",
       "      <td>0.866667</td>\n",
       "      <td>0.047879</td>\n",
       "      <td>0.319121</td>\n",
       "      <td>0.047879</td>\n",
       "      <td>0.319121</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>KNeighborsDist</td>\n",
       "      <td>0.633333</td>\n",
       "      <td>0.000982</td>\n",
       "      <td>0.007527</td>\n",
       "      <td>0.000982</td>\n",
       "      <td>0.007527</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>KNeighborsUnif</td>\n",
       "      <td>0.366667</td>\n",
       "      <td>0.001968</td>\n",
       "      <td>2.219379</td>\n",
       "      <td>0.001968</td>\n",
       "      <td>2.219379</td>\n",
       "      <td>1</td>\n",
       "      <td>True</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                  model  score_val  pred_time_val  fit_time  \\\n",
       "0         LightGBMLarge   0.966667       0.002011  0.261748   \n",
       "1   WeightedEnsemble_L2   0.966667       0.002989  0.440563   \n",
       "2              CatBoost   0.900000       0.003466  1.098149   \n",
       "3       NeuralNetFastAI   0.900000       0.007771  1.824703   \n",
       "4        NeuralNetTorch   0.900000       0.014041  0.907093   \n",
       "5        ExtraTreesGini   0.900000       0.039440  0.316139   \n",
       "6      RandomForestGini   0.900000       0.053123  0.424222   \n",
       "7            LightGBMXT   0.866667       0.000998  0.694706   \n",
       "8              LightGBM   0.866667       0.002006  0.196532   \n",
       "9               XGBoost   0.866667       0.004032  0.182862   \n",
       "10     RandomForestEntr   0.866667       0.039455  0.396508   \n",
       "11       ExtraTreesEntr   0.866667       0.047879  0.319121   \n",
       "12       KNeighborsDist   0.633333       0.000982  0.007527   \n",
       "13       KNeighborsUnif   0.366667       0.001968  2.219379   \n",
       "\n",
       "    pred_time_val_marginal  fit_time_marginal  stack_level  can_infer  \\\n",
       "0                 0.002011           0.261748            1       True   \n",
       "1                 0.000978           0.178815            2       True   \n",
       "2                 0.003466           1.098149            1       True   \n",
       "3                 0.007771           1.824703            1       True   \n",
       "4                 0.014041           0.907093            1       True   \n",
       "5                 0.039440           0.316139            1       True   \n",
       "6                 0.053123           0.424222            1       True   \n",
       "7                 0.000998           0.694706            1       True   \n",
       "8                 0.002006           0.196532            1       True   \n",
       "9                 0.004032           0.182862            1       True   \n",
       "10                0.039455           0.396508            1       True   \n",
       "11                0.047879           0.319121            1       True   \n",
       "12                0.000982           0.007527            1       True   \n",
       "13                0.001968           2.219379            1       True   \n",
       "\n",
       "    fit_order  \n",
       "0          13  \n",
       "1          14  \n",
       "2           7  \n",
       "3          10  \n",
       "4          12  \n",
       "5           8  \n",
       "6           5  \n",
       "7           3  \n",
       "8           4  \n",
       "9          11  \n",
       "10          6  \n",
       "11          9  \n",
       "12          2  \n",
       "13          1  "
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "predictor.leaderboard()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'path': 'AutogluonModels\\\\ag-20230923_044111\\\\',\n",
       " 'label': 'class',\n",
       " 'random_state': 0,\n",
       " 'version': '0.7.0',\n",
       " 'features': ['年龄',\n",
       "  '性别',\n",
       "  '脑出血前mRS评分',\n",
       "  '高血压病史',\n",
       "  '卒中病史',\n",
       "  '糖尿病史',\n",
       "  '房颤史',\n",
       "  '冠心病史',\n",
       "  '吸烟史',\n",
       "  '饮酒史',\n",
       "  '发病到首次影像检查时间间隔',\n",
       "  '脑室引流',\n",
       "  '止血治疗',\n",
       "  '降颅压治疗',\n",
       "  '降压治疗',\n",
       "  '镇静、镇痛治疗',\n",
       "  '止吐护胃',\n",
       "  '营养神经',\n",
       "  'HM_volume',\n",
       "  'HM_ACA_R_Ratio',\n",
       "  'HM_MCA_R_Ratio',\n",
       "  'HM_PCA_R_Ratio',\n",
       "  'HM_Pons_Medulla_R_Ratio',\n",
       "  'HM_Cerebellum_R_Ratio',\n",
       "  'HM_ACA_L_Ratio',\n",
       "  'HM_MCA_L_Ratio',\n",
       "  'HM_PCA_L_Ratio',\n",
       "  'HM_Pons_Medulla_L_Ratio',\n",
       "  'HM_Cerebellum_L_Ratio',\n",
       "  'ED_volume',\n",
       "  'ED_ACA_R_Ratio',\n",
       "  'ED_MCA_R_Ratio',\n",
       "  'ED_PCA_R_Ratio',\n",
       "  'ED_Pons_Medulla_R_Ratio',\n",
       "  'ED_Cerebellum_R_Ratio',\n",
       "  'ED_ACA_L_Ratio',\n",
       "  'ED_MCA_L_Ratio',\n",
       "  'ED_PCA_L_Ratio',\n",
       "  'ED_Pons_Medulla_L_Ratio',\n",
       "  'ED_Cerebellum_L_Ratio',\n",
       "  'original_shape_Elongation_x',\n",
       "  'original_shape_Flatness_x',\n",
       "  'original_shape_LeastAxisLength_x',\n",
       "  'original_shape_MajorAxisLength_x',\n",
       "  'original_shape_Maximum2DDiameterColumn_x',\n",
       "  'original_shape_Maximum2DDiameterRow_x',\n",
       "  'original_shape_Maximum2DDiameterSlice_x',\n",
       "  'original_shape_Maximum3DDiameter_x',\n",
       "  'original_shape_MeshVolume_x',\n",
       "  'original_shape_MinorAxisLength_x',\n",
       "  'original_shape_Sphericity_x',\n",
       "  'original_shape_SurfaceArea_x',\n",
       "  'original_shape_SurfaceVolumeRatio_x',\n",
       "  'original_shape_VoxelVolume_x',\n",
       "  'NCCT_original_firstorder_10Percentile_x',\n",
       "  'NCCT_original_firstorder_90Percentile_x',\n",
       "  'NCCT_original_firstorder_Energy_x',\n",
       "  'NCCT_original_firstorder_Entropy_x',\n",
       "  'NCCT_original_firstorder_InterquartileRange_x',\n",
       "  'NCCT_original_firstorder_Kurtosis_x',\n",
       "  'NCCT_original_firstorder_Maximum_x',\n",
       "  'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "  'NCCT_original_firstorder_Mean_x',\n",
       "  'NCCT_original_firstorder_Median_x',\n",
       "  'NCCT_original_firstorder_Minimum_x',\n",
       "  'NCCT_original_firstorder_Range_x',\n",
       "  'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "  'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "  'NCCT_original_firstorder_Skewness_x',\n",
       "  'NCCT_original_firstorder_Uniformity_x',\n",
       "  'NCCT_original_firstorder_Variance_x',\n",
       "  'original_shape_Elongation_y',\n",
       "  'original_shape_Flatness_y',\n",
       "  'original_shape_LeastAxisLength_y',\n",
       "  'original_shape_MajorAxisLength_y',\n",
       "  'original_shape_Maximum2DDiameterColumn_y',\n",
       "  'original_shape_Maximum2DDiameterRow_y',\n",
       "  'original_shape_Maximum2DDiameterSlice_y',\n",
       "  'original_shape_Maximum3DDiameter_y',\n",
       "  'original_shape_MeshVolume_y',\n",
       "  'original_shape_MinorAxisLength_y',\n",
       "  'original_shape_Sphericity_y',\n",
       "  'original_shape_SurfaceArea_y',\n",
       "  'original_shape_SurfaceVolumeRatio_y',\n",
       "  'original_shape_VoxelVolume_y',\n",
       "  'NCCT_original_firstorder_10Percentile_y',\n",
       "  'NCCT_original_firstorder_90Percentile_y',\n",
       "  'NCCT_original_firstorder_Energy_y',\n",
       "  'NCCT_original_firstorder_Entropy_y',\n",
       "  'NCCT_original_firstorder_InterquartileRange_y',\n",
       "  'NCCT_original_firstorder_Kurtosis_y',\n",
       "  'NCCT_original_firstorder_Maximum_y',\n",
       "  'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "  'NCCT_original_firstorder_Mean_y',\n",
       "  'NCCT_original_firstorder_Median_y',\n",
       "  'NCCT_original_firstorder_Minimum_y',\n",
       "  'NCCT_original_firstorder_Range_y',\n",
       "  'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "  'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "  'NCCT_original_firstorder_Skewness_y',\n",
       "  'NCCT_original_firstorder_Uniformity_y',\n",
       "  'NCCT_original_firstorder_Variance_y',\n",
       "  '高压',\n",
       "  '低压'],\n",
       " 'feature_metadata_in': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cde8733c40>,\n",
       " 'time_fit_preprocessing': 0.07637143135070801,\n",
       " 'time_fit_training': 9.42396330833435,\n",
       " 'time_fit_total': 9.500334739685059,\n",
       " 'time_limit': None,\n",
       " 'time_train_start': 1695444071.3732486,\n",
       " 'num_rows_train': 116,\n",
       " 'num_cols_train': 104,\n",
       " 'num_rows_val': 30,\n",
       " 'num_classes': 2,\n",
       " 'problem_type': 'binary',\n",
       " 'eval_metric': 'accuracy',\n",
       " 'best_model': 'WeightedEnsemble_L2',\n",
       " 'best_model_score_val': 0.9666666666666667,\n",
       " 'best_model_stack_level': 2,\n",
       " 'num_models_trained': 14,\n",
       " 'num_bag_folds': 0,\n",
       " 'max_stack_level': 2,\n",
       " 'max_core_stack_level': 1,\n",
       " 'model_info': {'KNeighborsUnif': {'name': 'KNeighborsUnif',\n",
       "   'model_type': 'KNNModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 2.219379186630249,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0019676685333251953,\n",
       "   'val_score': 0.36666666666666664,\n",
       "   'hyperparameters': {'weights': 'uniform'},\n",
       "   'hyperparameters_fit': {},\n",
       "   'hyperparameters_nondefault': ['weights'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['int', 'float'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': ['bool'],\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 89,\n",
       "   'features': ['年龄',\n",
       "    '脑出血前mRS评分',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd8a699b50>,\n",
       "   'memory_size': 119773,\n",
       "   'compile_time': None},\n",
       "  'KNeighborsDist': {'name': 'KNeighborsDist',\n",
       "   'model_type': 'KNNModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.007526874542236328,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0009822845458984375,\n",
       "   'val_score': 0.6333333333333333,\n",
       "   'hyperparameters': {'weights': 'distance'},\n",
       "   'hyperparameters_fit': {},\n",
       "   'hyperparameters_nondefault': ['weights'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['int', 'float'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': ['bool'],\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 89,\n",
       "   'features': ['年龄',\n",
       "    '脑出血前mRS评分',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86ca0b50>,\n",
       "   'memory_size': 119774,\n",
       "   'compile_time': None},\n",
       "  'LightGBMXT': {'name': 'LightGBMXT',\n",
       "   'model_type': 'LGBModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.6947064399719238,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0009984970092773438,\n",
       "   'val_score': 0.8666666666666667,\n",
       "   'hyperparameters': {'learning_rate': 0.05, 'extra_trees': True},\n",
       "   'hyperparameters_fit': {'num_boost_round': 86},\n",
       "   'hyperparameters_nondefault': ['extra_trees'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86ca0610>,\n",
       "   'memory_size': 67545,\n",
       "   'compile_time': None},\n",
       "  'LightGBM': {'name': 'LightGBM',\n",
       "   'model_type': 'LGBModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.1965320110321045,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0020062923431396484,\n",
       "   'val_score': 0.8666666666666667,\n",
       "   'hyperparameters': {'learning_rate': 0.05},\n",
       "   'hyperparameters_fit': {'num_boost_round': 148},\n",
       "   'hyperparameters_nondefault': [],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86c95130>,\n",
       "   'memory_size': 107063,\n",
       "   'compile_time': None},\n",
       "  'RandomForestGini': {'name': 'RandomForestGini',\n",
       "   'model_type': 'RFModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.4242222309112549,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.05312347412109375,\n",
       "   'val_score': 0.9,\n",
       "   'hyperparameters': {'n_estimators': 300,\n",
       "    'max_leaf_nodes': 15000,\n",
       "    'n_jobs': -1,\n",
       "    'random_state': 0,\n",
       "    'bootstrap': True,\n",
       "    'criterion': 'gini'},\n",
       "   'hyperparameters_fit': {'n_estimators': 300},\n",
       "   'hyperparameters_nondefault': ['criterion'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86a30700>,\n",
       "   'memory_size': 728100,\n",
       "   'compile_time': None},\n",
       "  'RandomForestEntr': {'name': 'RandomForestEntr',\n",
       "   'model_type': 'RFModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.39650774002075195,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.03945517539978027,\n",
       "   'val_score': 0.8666666666666667,\n",
       "   'hyperparameters': {'n_estimators': 300,\n",
       "    'max_leaf_nodes': 15000,\n",
       "    'n_jobs': -1,\n",
       "    'random_state': 0,\n",
       "    'bootstrap': True,\n",
       "    'criterion': 'entropy'},\n",
       "   'hyperparameters_fit': {'n_estimators': 300},\n",
       "   'hyperparameters_nondefault': ['criterion'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86a307f0>,\n",
       "   'memory_size': 689061,\n",
       "   'compile_time': None},\n",
       "  'CatBoost': {'name': 'CatBoost',\n",
       "   'model_type': 'CatBoostModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 1.0981488227844238,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0034661293029785156,\n",
       "   'val_score': 0.9,\n",
       "   'hyperparameters': {'iterations': 10000,\n",
       "    'learning_rate': 0.05,\n",
       "    'random_seed': 0,\n",
       "    'allow_writing_files': False,\n",
       "    'eval_metric': 'Accuracy'},\n",
       "   'hyperparameters_fit': {'iterations': 33},\n",
       "   'hyperparameters_nondefault': [],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86c8b100>,\n",
       "   'memory_size': 60174,\n",
       "   'compile_time': None},\n",
       "  'ExtraTreesGini': {'name': 'ExtraTreesGini',\n",
       "   'model_type': 'XTModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.31613945960998535,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.039440155029296875,\n",
       "   'val_score': 0.9,\n",
       "   'hyperparameters': {'n_estimators': 300,\n",
       "    'max_leaf_nodes': 15000,\n",
       "    'n_jobs': -1,\n",
       "    'random_state': 0,\n",
       "    'bootstrap': True,\n",
       "    'criterion': 'gini'},\n",
       "   'hyperparameters_fit': {'n_estimators': 300},\n",
       "   'hyperparameters_nondefault': ['criterion'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd8aedb910>,\n",
       "   'memory_size': 1110877,\n",
       "   'compile_time': None},\n",
       "  'ExtraTreesEntr': {'name': 'ExtraTreesEntr',\n",
       "   'model_type': 'XTModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.3191213607788086,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.04787874221801758,\n",
       "   'val_score': 0.8666666666666667,\n",
       "   'hyperparameters': {'n_estimators': 300,\n",
       "    'max_leaf_nodes': 15000,\n",
       "    'n_jobs': -1,\n",
       "    'random_state': 0,\n",
       "    'bootstrap': True,\n",
       "    'criterion': 'entropy'},\n",
       "   'hyperparameters_fit': {'n_estimators': 300},\n",
       "   'hyperparameters_nondefault': ['criterion'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd8aedb700>,\n",
       "   'memory_size': 1111893,\n",
       "   'compile_time': None},\n",
       "  'NeuralNetFastAI': {'name': 'NeuralNetFastAI',\n",
       "   'model_type': 'NNFastAiTabularModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 1.8247029781341553,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.007771015167236328,\n",
       "   'val_score': 0.9,\n",
       "   'hyperparameters': {'layers': None,\n",
       "    'emb_drop': 0.1,\n",
       "    'ps': 0.1,\n",
       "    'bs': 'auto',\n",
       "    'lr': 0.01,\n",
       "    'epochs': 'auto',\n",
       "    'early.stopping.min_delta': 0.0001,\n",
       "    'early.stopping.patience': 20,\n",
       "    'smoothing': 0.0},\n",
       "   'hyperparameters_fit': {'epochs': 30, 'best_epoch': 5},\n",
       "   'hyperparameters_nondefault': [],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': ['text_ngram', 'text_as_category'],\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86c25340>,\n",
       "   'memory_size': 220062,\n",
       "   'compile_time': None},\n",
       "  'XGBoost': {'name': 'XGBoost',\n",
       "   'model_type': 'XGBoostModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.1828615665435791,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.004031658172607422,\n",
       "   'val_score': 0.8666666666666667,\n",
       "   'hyperparameters': {'n_estimators': 10000,\n",
       "    'learning_rate': 0.1,\n",
       "    'n_jobs': -1,\n",
       "    'proc.max_category_levels': 100,\n",
       "    'objective': 'binary:logistic',\n",
       "    'booster': 'gbtree'},\n",
       "   'hyperparameters_fit': {'n_estimators': 55},\n",
       "   'hyperparameters_nondefault': [],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86d002e0>,\n",
       "   'memory_size': 315800,\n",
       "   'compile_time': None},\n",
       "  'NeuralNetTorch': {'name': 'NeuralNetTorch',\n",
       "   'model_type': 'TabularNeuralNetTorchModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.9070932865142822,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.014040708541870117,\n",
       "   'val_score': 0.9,\n",
       "   'hyperparameters': {'num_epochs': 500,\n",
       "    'epochs_wo_improve': 20,\n",
       "    'activation': 'relu',\n",
       "    'embedding_size_factor': 1.0,\n",
       "    'embed_exponent': 0.56,\n",
       "    'max_embedding_dim': 100,\n",
       "    'y_range': None,\n",
       "    'y_range_extend': 0.05,\n",
       "    'dropout_prob': 0.1,\n",
       "    'optimizer': 'adam',\n",
       "    'learning_rate': 0.0003,\n",
       "    'weight_decay': 1e-06,\n",
       "    'proc.embed_min_categories': 4,\n",
       "    'proc.impute_strategy': 'median',\n",
       "    'proc.max_category_levels': 100,\n",
       "    'proc.skew_threshold': 0.99,\n",
       "    'use_ngram_features': False,\n",
       "    'num_layers': 4,\n",
       "    'hidden_size': 128,\n",
       "    'max_batch_size': 512,\n",
       "    'use_batchnorm': False,\n",
       "    'loss_function': 'auto'},\n",
       "   'hyperparameters_fit': {'batch_size': 32, 'num_epochs': 34},\n",
       "   'hyperparameters_nondefault': [],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': ['text_ngram', 'text_as_category'],\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86d0be20>,\n",
       "   'memory_size': 315264,\n",
       "   'compile_time': None},\n",
       "  'LightGBMLarge': {'name': 'LightGBMLarge',\n",
       "   'model_type': 'LGBModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.2617480754852295,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.002010822296142578,\n",
       "   'val_score': 0.9666666666666667,\n",
       "   'hyperparameters': {'learning_rate': 0.03,\n",
       "    'num_leaves': 128,\n",
       "    'feature_fraction': 0.9,\n",
       "    'min_data_in_leaf': 5},\n",
       "   'hyperparameters_fit': {'num_boost_round': 17},\n",
       "   'hyperparameters_nondefault': ['learning_rate',\n",
       "    'num_leaves',\n",
       "    'feature_fraction',\n",
       "    'min_data_in_leaf'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': ['bool', 'int', 'float', 'category'],\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None},\n",
       "   'num_features': 104,\n",
       "   'features': ['年龄',\n",
       "    '性别',\n",
       "    '脑出血前mRS评分',\n",
       "    '高血压病史',\n",
       "    '卒中病史',\n",
       "    '糖尿病史',\n",
       "    '房颤史',\n",
       "    '冠心病史',\n",
       "    '吸烟史',\n",
       "    '饮酒史',\n",
       "    '发病到首次影像检查时间间隔',\n",
       "    '脑室引流',\n",
       "    '止血治疗',\n",
       "    '降颅压治疗',\n",
       "    '降压治疗',\n",
       "    '镇静、镇痛治疗',\n",
       "    '止吐护胃',\n",
       "    '营养神经',\n",
       "    'HM_volume',\n",
       "    'HM_ACA_R_Ratio',\n",
       "    'HM_MCA_R_Ratio',\n",
       "    'HM_PCA_R_Ratio',\n",
       "    'HM_Pons_Medulla_R_Ratio',\n",
       "    'HM_Cerebellum_R_Ratio',\n",
       "    'HM_ACA_L_Ratio',\n",
       "    'HM_MCA_L_Ratio',\n",
       "    'HM_PCA_L_Ratio',\n",
       "    'HM_Pons_Medulla_L_Ratio',\n",
       "    'HM_Cerebellum_L_Ratio',\n",
       "    'ED_volume',\n",
       "    'ED_ACA_R_Ratio',\n",
       "    'ED_MCA_R_Ratio',\n",
       "    'ED_PCA_R_Ratio',\n",
       "    'ED_Pons_Medulla_R_Ratio',\n",
       "    'ED_Cerebellum_R_Ratio',\n",
       "    'ED_ACA_L_Ratio',\n",
       "    'ED_MCA_L_Ratio',\n",
       "    'ED_PCA_L_Ratio',\n",
       "    'ED_Pons_Medulla_L_Ratio',\n",
       "    'ED_Cerebellum_L_Ratio',\n",
       "    'original_shape_Elongation_x',\n",
       "    'original_shape_Flatness_x',\n",
       "    'original_shape_LeastAxisLength_x',\n",
       "    'original_shape_MajorAxisLength_x',\n",
       "    'original_shape_Maximum2DDiameterColumn_x',\n",
       "    'original_shape_Maximum2DDiameterRow_x',\n",
       "    'original_shape_Maximum2DDiameterSlice_x',\n",
       "    'original_shape_Maximum3DDiameter_x',\n",
       "    'original_shape_MeshVolume_x',\n",
       "    'original_shape_MinorAxisLength_x',\n",
       "    'original_shape_Sphericity_x',\n",
       "    'original_shape_SurfaceArea_x',\n",
       "    'original_shape_SurfaceVolumeRatio_x',\n",
       "    'original_shape_VoxelVolume_x',\n",
       "    'NCCT_original_firstorder_10Percentile_x',\n",
       "    'NCCT_original_firstorder_90Percentile_x',\n",
       "    'NCCT_original_firstorder_Energy_x',\n",
       "    'NCCT_original_firstorder_Entropy_x',\n",
       "    'NCCT_original_firstorder_InterquartileRange_x',\n",
       "    'NCCT_original_firstorder_Kurtosis_x',\n",
       "    'NCCT_original_firstorder_Maximum_x',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_Mean_x',\n",
       "    'NCCT_original_firstorder_Median_x',\n",
       "    'NCCT_original_firstorder_Minimum_x',\n",
       "    'NCCT_original_firstorder_Range_x',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_x',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_x',\n",
       "    'NCCT_original_firstorder_Skewness_x',\n",
       "    'NCCT_original_firstorder_Uniformity_x',\n",
       "    'NCCT_original_firstorder_Variance_x',\n",
       "    'original_shape_Elongation_y',\n",
       "    'original_shape_Flatness_y',\n",
       "    'original_shape_LeastAxisLength_y',\n",
       "    'original_shape_MajorAxisLength_y',\n",
       "    'original_shape_Maximum2DDiameterColumn_y',\n",
       "    'original_shape_Maximum2DDiameterRow_y',\n",
       "    'original_shape_Maximum2DDiameterSlice_y',\n",
       "    'original_shape_Maximum3DDiameter_y',\n",
       "    'original_shape_MeshVolume_y',\n",
       "    'original_shape_MinorAxisLength_y',\n",
       "    'original_shape_Sphericity_y',\n",
       "    'original_shape_SurfaceArea_y',\n",
       "    'original_shape_SurfaceVolumeRatio_y',\n",
       "    'original_shape_VoxelVolume_y',\n",
       "    'NCCT_original_firstorder_10Percentile_y',\n",
       "    'NCCT_original_firstorder_90Percentile_y',\n",
       "    'NCCT_original_firstorder_Energy_y',\n",
       "    'NCCT_original_firstorder_Entropy_y',\n",
       "    'NCCT_original_firstorder_InterquartileRange_y',\n",
       "    'NCCT_original_firstorder_Kurtosis_y',\n",
       "    'NCCT_original_firstorder_Maximum_y',\n",
       "    'NCCT_original_firstorder_MeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_Mean_y',\n",
       "    'NCCT_original_firstorder_Median_y',\n",
       "    'NCCT_original_firstorder_Minimum_y',\n",
       "    'NCCT_original_firstorder_Range_y',\n",
       "    'NCCT_original_firstorder_RobustMeanAbsoluteDeviation_y',\n",
       "    'NCCT_original_firstorder_RootMeanSquared_y',\n",
       "    'NCCT_original_firstorder_Skewness_y',\n",
       "    'NCCT_original_firstorder_Uniformity_y',\n",
       "    'NCCT_original_firstorder_Variance_y',\n",
       "    '高压',\n",
       "    '低压'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd86c05a00>,\n",
       "   'memory_size': 54854,\n",
       "   'compile_time': None},\n",
       "  'WeightedEnsemble_L2': {'name': 'WeightedEnsemble_L2',\n",
       "   'model_type': 'WeightedEnsembleModel',\n",
       "   'problem_type': 'binary',\n",
       "   'eval_metric': 'accuracy',\n",
       "   'stopping_metric': 'accuracy',\n",
       "   'fit_time': 0.17881512641906738,\n",
       "   'num_classes': 2,\n",
       "   'quantile_levels': None,\n",
       "   'predict_time': 0.0009777545928955078,\n",
       "   'val_score': 0.9666666666666667,\n",
       "   'hyperparameters': {'use_orig_features': False,\n",
       "    'max_base_models': 25,\n",
       "    'max_base_models_per_type': 5,\n",
       "    'save_bag_folds': True},\n",
       "   'hyperparameters_fit': {},\n",
       "   'hyperparameters_nondefault': ['save_bag_folds'],\n",
       "   'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "    'max_time_limit_ratio': 1.0,\n",
       "    'max_time_limit': None,\n",
       "    'min_time_limit': 0,\n",
       "    'valid_raw_types': None,\n",
       "    'valid_special_types': None,\n",
       "    'ignored_type_group_special': None,\n",
       "    'ignored_type_group_raw': None,\n",
       "    'get_features_kwargs': None,\n",
       "    'get_features_kwargs_extra': None,\n",
       "    'predict_1_batch_size': None,\n",
       "    'temperature_scalar': None,\n",
       "    'drop_unique': False},\n",
       "   'num_features': 1,\n",
       "   'features': ['LightGBMLarge'],\n",
       "   'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd8a6d87f0>,\n",
       "   'memory_size': 4507,\n",
       "   'compile_time': None,\n",
       "   'bagged_info': {'child_model_type': 'GreedyWeightedEnsembleModel',\n",
       "    'num_child_models': 1,\n",
       "    'child_model_names': ['S1F1'],\n",
       "    '_n_repeats': 1,\n",
       "    '_k_per_n_repeat': [1],\n",
       "    '_random_state': 2,\n",
       "    'low_memory': False,\n",
       "    'bagged_mode': False,\n",
       "    'max_memory_size': 4507,\n",
       "    'min_memory_size': 4507,\n",
       "    'child_hyperparameters': {'ensemble_size': 100},\n",
       "    'child_hyperparameters_fit': {'ensemble_size': 1},\n",
       "    'child_ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "     'max_time_limit_ratio': 1.0,\n",
       "     'max_time_limit': None,\n",
       "     'min_time_limit': 0,\n",
       "     'valid_raw_types': None,\n",
       "     'valid_special_types': None,\n",
       "     'ignored_type_group_special': None,\n",
       "     'ignored_type_group_raw': None,\n",
       "     'get_features_kwargs': None,\n",
       "     'get_features_kwargs_extra': None,\n",
       "     'predict_1_batch_size': None,\n",
       "     'temperature_scalar': None,\n",
       "     'drop_unique': False}},\n",
       "   'stacker_info': {'num_base_models': 1,\n",
       "    'base_model_names': ['LightGBMLarge']},\n",
       "   'children_info': {'S1F1': {'name': 'S1F1',\n",
       "     'model_type': 'GreedyWeightedEnsembleModel',\n",
       "     'problem_type': 'binary',\n",
       "     'eval_metric': 'accuracy',\n",
       "     'stopping_metric': 'accuracy',\n",
       "     'fit_time': 0.17881512641906738,\n",
       "     'num_classes': 2,\n",
       "     'quantile_levels': None,\n",
       "     'predict_time': None,\n",
       "     'val_score': None,\n",
       "     'hyperparameters': {'ensemble_size': 100},\n",
       "     'hyperparameters_fit': {'ensemble_size': 1},\n",
       "     'hyperparameters_nondefault': [],\n",
       "     'ag_args_fit': {'max_memory_usage_ratio': 1.0,\n",
       "      'max_time_limit_ratio': 1.0,\n",
       "      'max_time_limit': None,\n",
       "      'min_time_limit': 0,\n",
       "      'valid_raw_types': None,\n",
       "      'valid_special_types': None,\n",
       "      'ignored_type_group_special': None,\n",
       "      'ignored_type_group_raw': None,\n",
       "      'get_features_kwargs': None,\n",
       "      'get_features_kwargs_extra': None,\n",
       "      'predict_1_batch_size': None,\n",
       "      'temperature_scalar': None,\n",
       "      'drop_unique': False},\n",
       "     'num_features': 1,\n",
       "     'features': ['LightGBMLarge'],\n",
       "     'feature_metadata': <autogluon.common.features.feature_metadata.FeatureMetadata at 0x1cd8a6d8a60>,\n",
       "     'memory_size': 5163,\n",
       "     'compile_time': None,\n",
       "     'model_weights': {'LightGBMLarge': 1.0}}}}}}"
      ]
     },
     "execution_count": 26,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "predictor.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'NN_TORCH': {},\n",
       " 'GBM': [{'extra_trees': True, 'ag_args': {'name_suffix': 'XT'}},\n",
       "  {},\n",
       "  'GBMLarge'],\n",
       " 'CAT': {},\n",
       " 'XGB': {},\n",
       " 'FASTAI': {},\n",
       " 'RF': [{'criterion': 'gini',\n",
       "   'ag_args': {'name_suffix': 'Gini',\n",
       "    'problem_types': ['binary', 'multiclass']}},\n",
       "  {'criterion': 'entropy',\n",
       "   'ag_args': {'name_suffix': 'Entr',\n",
       "    'problem_types': ['binary', 'multiclass']}},\n",
       "  {'criterion': 'squared_error',\n",
       "   'ag_args': {'name_suffix': 'MSE',\n",
       "    'problem_types': ['regression', 'quantile']}}],\n",
       " 'XT': [{'criterion': 'gini',\n",
       "   'ag_args': {'name_suffix': 'Gini',\n",
       "    'problem_types': ['binary', 'multiclass']}},\n",
       "  {'criterion': 'entropy',\n",
       "   'ag_args': {'name_suffix': 'Entr',\n",
       "    'problem_types': ['binary', 'multiclass']}},\n",
       "  {'criterion': 'squared_error',\n",
       "   'ag_args': {'name_suffix': 'MSE',\n",
       "    'problem_types': ['regression', 'quantile']}}],\n",
       " 'KNN': [{'weights': 'uniform', 'ag_args': {'name_suffix': 'Unif'}},\n",
       "  {'weights': 'distance', 'ag_args': {'name_suffix': 'Dist'}}]}"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "predictor.fit_hyperparameters_"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "base",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.12"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
